'''yaml相关操作'''
import os

import yaml

# 写入yaml文件
def write_yaml(filename, data):
    with open(os.getcwd() + "/data/" + filename, mode='a+', encoding='utf-8') as f:
        yaml.safe_dump(data, stream=f)


# 读取yaml文件
def read_yaml(filename, key):
    with open(os.getcwd() + "/data/" + filename, mode='r', encoding='utf-8') as f:
        data =  yaml.safe_load(f)
        return data[key]

# 清空yaml文件
def clear_yaml(filename):
    with open(os.getcwd() + "/data/" + filename, mode='w', encoding='utf-8') as f:
        f.truncate()